package cn.aposoft.mq.kafka.admin.topic;

import org.apache.kafka.clients.admin.Admin;
import org.apache.kafka.clients.admin.ListPartitionReassignmentsResult;
import org.apache.kafka.clients.admin.PartitionReassignment;
import org.apache.kafka.clients.producer.ProducerConfig;
import org.apache.kafka.common.KafkaFuture;
import org.apache.kafka.common.TopicPartition;

import java.util.HashSet;
import java.util.Map;
import java.util.Properties;
import java.util.Set;
import java.util.concurrent.ExecutionException;

public class ListPartitionReassignments {

    public static void main(String[] args) {
        Properties p = new Properties();
        p.put(ProducerConfig.BOOTSTRAP_SERVERS_CONFIG, "aposoft.cn:10091,aposoft.cn:10092,aposoft.cn:10093");// kafka地址，多个地址用逗号分割
        try (Admin admin = Admin.create(p);) {
            Set<TopicPartition> topicPartitionOffsets = new HashSet<>();
            TopicPartition partition = new TopicPartition("test-new", 0);
            topicPartitionOffsets.add(partition);
            ListPartitionReassignmentsResult result = admin.listPartitionReassignments(topicPartitionOffsets);
            KafkaFuture<Map<TopicPartition, PartitionReassignment>> future = result.reassignments();
            System.out.println(future.get());
        } catch (
                ExecutionException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}